版本:anaconda=4.9.2 python3.6 pytorch=1.4 cuda=10.1
情況說明:
因為之前安裝過tensorflow gpu版的,cuda10和cuda10.1都裝過了,是以本次記錄不涉及cuda和cudnn的安裝配置
提供tensorflow-gpu 以及 cuda的安裝參考教程 + 教程2 + 自己的安裝記錄(可用來參考)
本次安裝參考教程:此教程也涉及了cuda的安裝,講的也很清楚,隻想裝pytorch,看這個就夠了
具體安裝過程
進入終端 —— 建立虛拟環境conda create -n pytorch1.4 python3.6 —— 激活環境 —— 在官網找到對應語句 【我的是 conda install pytorch1.4.0 torchvision0.5.0 cudatoolkit=10.1 -c pytorch】(具體見圖1)—— 完成 —— 測試:還是在pytorch1.4環境下,輸入python之後(正常應該顯示的版本為python3.6.13)繼續輸入
import torch
print(torch.version)
print(torch.cuda.is_available())
【結果見圖2】
圖1
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLiAzNfRHLGZkRGZkRfJ3bs92YsYTMfVmepNHL90TQjZmQuJGashVYohmMMBjVtJWd0ckW65UbM5WOHJWa5kHT20ESjBjUIF2X0hXZ0xCMx81dvRWYoNHLrdEZwZ1Rh5WNXp1bwNjW1ZUba9VZwlHdssmch1mclRXY39CXldWYtlWPzNXZj9mcw1ycz9WL49zZuBnL4UzNxUDMyATM5ITNwEjMwIzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
圖2
其他包的安裝