python版本:3.7
CUDA版本:10.0
安装相应的PyTorch的cmd命令:
conda install pytorch torchvision cuda100 -c pytorch
安装速度还是挺快的(不到10分钟)
注意:安装前以管理员身份运行cmd
测试程序:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
a = torch.randn(2,3)
print(a)
运行结果:
更新:
再次安装时,使用原来的方法遇到了问题,新的解决方法如下:
python版本:3.7
CUDA版本:10.0
在 INSTALLING PREVIOUS VERSIONS OF PYTORCH 搜索“CUDA 10.0”:
可以发现,需要安装PyTorch 1.2.0,但使用网站给的安装命令:
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
出现了问题
解决办法:
到https://download.pytorch.org/whl/cu100/torch_stable.html
下载torch-1.2.0-cp37-cp37m-win_amd64.whl
下载完后在下载到的路径下运行:
pip install torch-1.2.0-cp37-cp37m-win_amd64.whl
即安装完成。
安装torchvision同理
相关文章:
INSTALLING PREVIOUS VERSIONS OF PYTORCH
How to Install PyTorch with CUDA 10.0