CUDA升级（在conda虚拟环境中）

CUDA版本不够，升级又太复杂，应该怎么办呢？

服务器cuda的版本是9.0，只能安装很低版本的pytorch，查了查怎么升级cuda，感觉好复杂，幸好可以在虚拟环境中安装cuda 10.1，省了不少力气。

在虚拟环境中安装新版本的cuda

1、创建虚拟环境。（注意cuda10为虚拟环境的名字）

conda create -n cuda10 python=3.7

2、进入环境后，安装cuda和cudnn

conda activate cuda10
conda install cudatoolkit=10.1
conda install cudnn=7.6.5

这样就结束啦，可以去pytorch的官网安装对应版本的torch啦！