1.#检查cuda的安装情况
nvcc --version
2.将cudnn中的库文件移动到cuda的文件夹中
3.#检查cudnn,跑demo程序,两个PASS
4.#cuda9.0
conda install pytorch1.0.0 torchvision0.2.1 cuda90
5.#添加源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
6.安装pytorch
conda install --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ pytorch=1.0.0 torchvision=0.2.1 cuda90 -c pytorch
win10+cuda9.0+cudnn7.6.5 for cuda9.0+torch1.0
最新推荐文章于 2022-05-08 20:47:04 发布