天天看點

基于Win10的anaconda下pytorch gpu版安裝

版本: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

基于Win10的anaconda下pytorch gpu版安裝

圖2

基于Win10的anaconda下pytorch gpu版安裝

其他包的安裝

基于Win10的anaconda下pytorch gpu版安裝
基于Win10的anaconda下pytorch gpu版安裝