环境配置步骤:
cuda驱动—cuda10—cuDNN—anaconda3—pytorch
cuda10 + cuDNN10安装:
参考:cuda安装
1、卸载原驱动
2、安装对应cuda版本驱动
安装驱动可能的问题解决:
问题1
问题2
3、安装cuda
下载后的文件为cuda_10.0.130_410.48_linux.run,为文件赋予执行权限
chmod a+x cuda_10.0.130_410.48_linux.run
进行安装,其中download是下载后cuda所在的目录
sudo sh cuda_10.0.130_410.48_linux.run
修改环境变量:修改环境变量
sudo gedit ~/.bashrc
//添加三个路径
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME