PyTorch笔记 安装

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