linux 安装cuda10.1的 pytorch python3.7
时间: 2023-08-28 20:17:41 浏览: 155
pytorch-1.7.0.dev20200626-py3.6_cuda10.1.243_cudnn7.6.3_0.tar.bz2
安装 CUDA 10.1
1. 下载 CUDA 10.1 安装包
在 NVIDIA 官网上下载 CUDA 10.1 安装包。选择适合你的操作系统和显卡型号的版本。下载地址为 https://developer.nvidia.com/cuda-10.1-download-archive-base
2. 安装 CUDA 10.1
解压下载好的文件,进入目录,运行以下命令安装:
```
$ sudo sh cuda_10.1.243_418.87.00_linux.run
```
根据提示进行安装,选择合适的选项即可。
3. 配置环境变量
安装完成后,需要配置环境变量。在 ~/.bashrc 文件中添加以下内容:
```
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
```
然后运行以下命令使其生效:
```
$ source ~/.bashrc
```
安装 PyTorch
1. 安装依赖
PyTorch 依赖于以下软件包:
- Python 3.5 或以上版本
- NumPy
- setuptools
- wheel
- pyyaml
- cffi
- ninja
- mkl
- mkl-include
使用以下命令安装这些依赖:
```
$ sudo apt-get install python3-dev python3-pip
$ pip3 install numpy setuptools wheel pyyaml cffi ninja
```
2. 安装 PyTorch
使用以下命令安装 PyTorch:
```
$ pip3 install torch torchvision
```
以上命令将会安装最新版本的 PyTorch 和 torchvision。
如果你需要安装特定版本的 PyTorch,可以使用以下命令:
```
$ pip3 install torch==1.4.0 torchvision==0.5.0
```
这将会安装 PyTorch 1.4.0 和 torchvision 0.5.0。
至此,你已经成功安装了 CUDA 10.1 和 PyTorch,并且可以在 Python 3.7 中使用它们了。
阅读全文