RTX3090 windows cuda11.0 cudnn pytorch nightly试用版

 

RTX3090到货了，花了一点时间重新装了cuda 和配了一下环境参考了这一篇文章：虽然标题写的是劝退文，但是他更新了说10.15号可以安装torch

https://blog.csdn.net/q1042960848/article/details/107856596

https://blog.csdn.net/lp1003/article/details/107240520/

1.下载cuda11.0

首先去NVIDA官网下载CUDA：

https://developer.nvidia.com/cuda-toolkit-archive

这里我下载的是现在最新的V11.0版本

-----

安装CUDA：

安装时选择：自定义安装-》只选择第一个-》一直到最后安装完成即可。主要是下面这一步的问题

这里可以参考：https://blog.csdn.net/shuiyixin/article/details/99935799

2.卸载cuda以前的版本

卸载cuda 我是在电脑里面的 控制面板--程序与安装里面--把与新版本重复的cuda都卸载了，还看了安装时间

然后再去环境路径中，把cuda10.0的路径删掉！这是时候一定要重启，这样才可以更新环境路径

2.下载cudnn

安装好后再在NVIDA官网上下载cuDNN，下载这个的时候需要注册。不需要翻墙也可以注册：

https://developer.nvidia.com/rdp/cudnn-download

这里我下载的是V8.0.1版本

一般来说下载的时候官网会根据做出的系统选项来推荐合适的CUDA以及cuDNN版本。

当然自己也可以根据本身电脑需求合理选择下载。

而cuDNN就是一些文件，将其中的文件复制到CUDA安装路径相应的目录下即可。不用担心会产生替换的问题。

3.卸载之前的pytorch

在conda uninstall pytorch 

4. 使用安装测试版本的pytroch 感谢知乎上的热心网友

pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu110/torch_nightly.html

5.然后在测试的时候有个小问题，大概是在import numpy的时候显示路径不对，大概要重新

conda uninstall numpy

conda install numpy 

就好啦！

import torch
print(torch.rand(3,3).cuda())