關于安裝pytorch的一些問題總結

• • 環境介紹
  • 安裝pytorch遇到的一些問題
  • pytorch安裝後測試是否可用
  • 一台電腦是否可以安裝不同版本的pytorch,cuda.....
  • RuntimeError: CUDA error: no kernel image is available for execution on the device
  • 檢視`torch`版本
  • 檢視自己電腦的GPU算力

文章目錄

• • 環境介紹
  • 安裝pytorch遇到的一些問題
  • pytorch安裝後測試是否可用
  • 一台電腦是否可以安裝不同版本的pytorch,cuda.....
  • RuntimeError: CUDA error: no kernel image is available for execution on the device
  • 檢視`torch`版本
  • 檢視自己電腦的GPU算力

環境介紹

window10
Python36虛拟環境
   python: 3.6.13
   cudatoolkit 10
   cudnn: 7
   pytorch: 1.2
   GPU：GeForce GT 730 算力：3.5
           

安裝pytorch遇到的一些問題

pytorch安裝後測試是否可用

import torch
 
print(torch.__version__) 
print(torch.cuda.is_available())
 
x=torch.randn(1)
if torch.cuda.is_available():
    device=torch.device("cuda")
    y=torch.ones_like(x,device=device)
    x=x.to(device)
    z=x+y
    print(z)
    print(z.to("cpu",torch.double))
           

上面這段代碼主要測試了

1. torch是否安裝成功
2. torch是否可以使用cuda加速
           

我猜肯定出現錯誤，出現錯誤可以接着往下看

一台電腦是否可以安裝不同版本的pytorch,cuda…

一台電腦可以安裝不同版本的

cuda,pytorch,python

,你隻需建立不同的虛拟環境就行了。

RuntimeError: CUDA error: no kernel image is available for execution on the device

當你出現這個錯誤時，不妨試試降低

pytorch

的版本。

我的台式電腦就出現了這種情況，我是這樣解決的，又切換了一個虛拟環境，安裝了較低版本的

cudatoolkit

,和

cudnn

,其實

cuda

和

cudnn

不是最主要的，但是如果你回退不到低版本的

pytorch

，可以這樣試試，下載下傳低版本的

cudatoolkit

和

cudnn

降低pytorch版本的方法直接安裝就行了，實在不行就在建立個虛拟環境

pip install torch==1.4.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
           

下面是

torch& torchvision

對應版本

檢視

torch

版本

在

anaconda prompt

激活環境，進入

python

,輸入下面的代碼

import torch
print(torch.__version__)
           

檢視自己電腦的GPU算力

首先找到

cuda

的安裝目錄，一般是下面的這個目錄

然後進入

cmd

,進入這個檔案夾，一次輸入圖檔中的檔案位址

上圖中的箭頭就是

GPU

的算力，同時也能檢視你的

CUDA

驅動器版本，

GPU

類型上面的紅框就是