windows10 python3.7环境安装:
建议创建一个新的python环境来使用pytorch
使用conda安装的命令:
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1
使用pip3安装的命令:
pip install torch==1.4.0 torchvision==0.5.0
下载cuda和cudnn:
cuda10.1下载网址https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
操作图:
最后在ide或者pycharm等编译器里面执行以下代码:
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
控制台显示:
1.4.0
10.1
True
表示安装成功
有什么问题吗??评论区指出