安装pytorch详细教程
·
1.去pytorch官网下载合适版本
直接在anaconda prompt上输入此命令，既可以安装，如果安装速度过慢，可以直接搜索cuda 10.2下载安装包安装。NVIDIA Developer
2.下载cudnn 10.2版本https://developer.nvidia.com/rdp/cudnn-download
3.安装
讲cudnn解压之后把其文件拷贝到cuda10.2中
4.验证其安装是否正确
#查看安装位置
set cuda
#查看版本号
nvcc -V
5.安装pytorch
查看已有的环境
conda info -e
conda create -n pytorch python=3.7
可用以上命令激活
conda activate pytorch_RSAN 
6.检查安装是否成功
测试代码如下：
import torch
print("hello {}".format(torch.__version__))
print(torch.cuda.is_available())返回torch的版本号和 True即可
免费领 150 小时云算力，进群参与显卡、AI PC 幸运抽奖
更多推荐
5
0- 0
已为社区贡献1条内容
所有评论(0)