環境說明:windows10 python3.6 anaconda3
1.按照進入pytorch官網https://pytorch.org/選擇正確的命令
它會自動爲你選擇一些配置,複製命令
我選擇的是:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
2.安裝出現報錯:
ERROR conda.core.link:_execute(502): An error occurred while uninstalling package 'defaults::conda-4.5.4-py27_0'.
OSError(13, 'Permission denied')
Attempting to roll back.
Rolling back transaction: done
OSError(13, 'Permission denied')
3.更新pip
pip install --user --upgrade pip
4.更改anconda權限,在文件夾下右鍵——屬性——安全
5.完成安裝: conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
6.測試
附註:
import torch
print(torch.version.cuda)